package com.base.service;

import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.base.entity.bo.TitleCategoryBo;
import com.base.entity.vo.TitleCategoryVo;
import com.base.mapper.TitleCategoryMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

@Service
public class TitleCategoryService {
    @Autowired
    TitleCategoryMapper titleCategoryMapper;

    public List titleCategoryList(TitleCategoryVo titleCategoryVo) {
        List<TitleCategoryBo> list = titleCategoryMapper.titleCategoryList(titleCategoryVo);
        return list;
    }

    public int updateTitleCategory(TitleCategoryVo titleCategoryVo) {
        TitleCategoryBo titleCategoryBo=new TitleCategoryBo();
        int row=0;
        for(int i=0;i<titleCategoryVo.getTitleHaveCategory().length;i++){
            titleCategoryBo.setUserFlag("1");
            titleCategoryBo.setId(titleCategoryVo.getTitleHaveCategory()[i]);
            row=titleCategoryMapper.update(titleCategoryBo,new UpdateWrapper<TitleCategoryBo>().eq("ID",titleCategoryBo.getId()));
        }
        for(int i=0;i<titleCategoryVo.getTitleOptionalCategory().length;i++){
            titleCategoryBo.setUserFlag("0");
            titleCategoryBo.setId(titleCategoryVo.getTitleOptionalCategory()[i]);
            row=titleCategoryMapper.update(titleCategoryBo,new UpdateWrapper<TitleCategoryBo>().eq("ID",titleCategoryBo.getId()));
        }
        return row;
    }
}
